from torchvision import datasets, transforms
# 数据集目录
data_dir = 'E:/Desktop/实验六/Monkeys'  # 替换为您的实际路径

# 使用 ImageFolder 读取数据
dataset = datasets.ImageFolder(root=data_dir, transform=transforms)

# 打印类别名称和编号
print("类别名称:", dataset.classes)  # 输出类别名称的列表
print("类别标签编号映射:", dataset.class_to_idx)  # 输出类别名称到标签编号的映射
